

# 训练相关
version = 'v9'
description = '10572 classes version:add birghtness enhance'
optimizer = 'sgd'
lr_scheduler = 'cosine-annealing'
end_epoch = 1000
lr = 0.003
lr_step = 10
log_out = fr"./{version}_log.log"
weights_to_load = None
weights_to_load = r'weights/v9_best.pth'
# weights_save_path = fr'weights/{version}'
batch_size = 64
data = 'CASIA-Webface'
use_gpu = True





# 网络相关
img_number = 490623
# img_number = 500
# img_number = 82425


img_cls = 10572
backbone_net = 'resnet18'
metric = 'insight_face'
# 超参
s = 30
m = 0.5
test_batchsize = 30
feature_dim = 512
datapath = r'E:\faces_webface_112x112'
lfw_root = r"C:\Users\2\Desktop\InsightFace\arcface-pytorch-master\data\Datasets\lfw-align-128"



